tensor([[[[[0.1500, 0.1500],
           [0.1500, 0.1500]],

          [[0.1500, 0.1500],
           [0.1500, 0.1500]]],


         [[[0.1500, 0.1500],
           [0.1500, 0.1500]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.4300, 0.4500],
           [0.4500, 0.4300]],

          [[0.4700, 0.4700],
           [0.4700, 0.4700]]],


         [[[0.4700, 0.4700],
           [0.4700, 0.4700]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.0700, 0.0400],
           [0.0400, 0.0700]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]],


         [[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[1.0000, 1.0000],
           [1.0000, 1.0000]]]],



        [[[[0.3500, 0.3600],
           [0.3600, 0.3500]],

          [[0.3800, 0.3800],
           [0.3800, 0.3800]]],


         [[[0.3800, 0.3800],
           [0.3800, 0.3800]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]]], grad_fn=<RoundBackward1>)